package com.lyf.common.utils;

import com.lyf.common.api.DataMaskType;
import org.apache.commons.lang3.StringUtils;

/**
 * @author yunfei.li
 * @date 2023/7/26 17:52
 */
public class MaskStrategy {

    public String process(String value,String maskType){
        DataMaskType dataMaskType = DataMaskType.valueOf(maskType);
        if (StringUtils.isNotBlank(value)) {
            switch (dataMaskType) {
                case CHINESE_NAME:
                    return chineseName(value);
                case ID_CARD:
                    return idCardNum(value);
                case MOBILE_PHONE:
                    return mobilePhone(value);
                case ADDRESS:
                    return address(value);
                case EMAIL:
                    return email(value);
                case BANK_CARD:
                    return bankCard(value);
                default:
                    return mobilePhone(value);
            }
        }
        return null;
    }
    /** * [中文姓名] 只显示第一个汉字，其他隐藏为2个星号<例子：李**> * * @param fullName 中文姓名 * @return 中文姓名 */
    private static String chineseName(String fullName) {
        if (StringUtils.isBlank(fullName)) {
            return "";
        }
        String name = StringUtils.left(fullName, 1);
        return StringUtils.rightPad(name, StringUtils.length(fullName), "*");
    }
    /** * [身份证号] 显示前六位，其他隐藏。共计18位或者15位。<例子：511623*************> * * @param id 身份证号 * @return 身份证号 */
    private static String idCardNum(String id) {
        if (StringUtils.isBlank(id)) {
            return "";
        }
        String num = StringUtils.left(id, 6);
        return StringUtils.leftPad(num, StringUtils.length(id), "*");
    }
    /** * [手机号码] 前三位，后四位，其他隐藏<例子:138******1234> * * @param num 手机号码 * @return String */
    private static String mobilePhone(String num) {
        if (StringUtils.isBlank(num)) {
            return "";
        }
        return StringUtils.left(num, 3).concat(StringUtils.removeStart(StringUtils.leftPad(StringUtils.right(num, 4), StringUtils.length(num), "*"), "***"));
    }
    /** * [地址] 只显示到省，不显示详细地址；我们要对个人信息增强保护<例子：北京市*******> * * @param address 地址 * @param sensitiveSize 敏感信息长度 * @return String */
    private static String address(String address) {
        if (StringUtils.isBlank(address)) {
            return "";
        }
        int length = StringUtils.length(address);
        return StringUtils.rightPad(StringUtils.left(address, 3), length, "*");
    }
    /** * [电子邮箱] 邮箱前缀仅显示第一个字母，前缀其他隐藏，用星号代替，@及后面的地址显示<例子:g**@163.com> * * @param email 电子邮箱 * @return String */
    private static String email(String email) {
        if (StringUtils.isBlank(email)) {
            return "";
        }
        int index = StringUtils.indexOf(email, "@");
        if (index <= 1) {
            return email;
        } else {
            return StringUtils.rightPad(StringUtils.left(email, 1), index, "*").concat(StringUtils.mid(email, index, StringUtils.length(email)));
        }
    }
    /** * [银行卡号] 前六位，后四位，其他用星号隐藏每位1个星号<例子:6222600**********1234> * * @param cardNum 银行卡号 * @return String */
    private static String bankCard(String cardNum) {
        if (StringUtils.isBlank(cardNum)) {
            return "";
        }
        return StringUtils.left(cardNum, 6).concat(StringUtils.removeStart(StringUtils.leftPad(StringUtils.right(cardNum, 4), StringUtils.length(cardNum), "*"), "******"));
    }
}
